Modify column values with list values when a condition is satisfied - PySpark